package com.westvalley.common.kit;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;

public class FileKit {
	
	/**
	 * 读取文件
	 * @param fileName
	 * @return
	 */
	public static File readFile(String fileName) {
		if (fileName != null) {
			File file = new File(fileName);
			if (file.exists()) {
				return file;
			}
		}
		return null;
	}
	
	/**
	 * 清空文件
	 * @param file
	 */
	public static void clearFile(File file){
		fileWriter(file,"",false);
	}
	
	/**
	 * 文件追加内容
	 * @param file
	 * @param context
	 */
	public static void fileAppend(File file,String context){
		fileWriter(file,context,true);
	}
	
	public static void fileWriter(File file,String context){
		fileWriter(file,context,false);
	}
	
	/**
	 * 文件内容写入
	 * @param file
	 * @param context
	 * @param append
	 */
	public static void fileWriter(File file,String context,boolean append){
		if(file != null && file.exists()){
			FileWriter fw;
			try {
				fw = new FileWriter(file,append);
				fw.write(context);
				fw.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
	
	/**
	 * 创建文件
	 * @param destFileName
	 * @return
	 */
	public static boolean createFile(String destFileName) {
        File file = new File(destFileName);
        if(file.exists()) {
            System.out.println("创建单个文件" + destFileName + "失败，目标文件已存在！");
            return false;
        }
        if (destFileName.endsWith(File.separator)) {
            System.out.println("创建单个文件" + destFileName + "失败，目标文件不能为目录！");
            return false;
        }
        //判断目标文件所在的目录是否存在
        if(!file.getParentFile().exists()) {
            //如果目标文件所在的目录不存在，则创建父目录
            System.out.println("目标文件所在目录不存在，准备创建它！");
            if(!file.getParentFile().mkdirs()) {
                System.out.println("创建目标文件所在目录失败！");
                return false;
            }
        }
        //创建目标文件
        try {
            if (file.createNewFile()) {
                System.out.println("创建单个文件" + destFileName + "成功！");
                return true;
            } else {
                System.out.println("创建单个文件" + destFileName + "失败！");
                return false;
            }
        } catch (IOException e) {
            e.printStackTrace();
            System.out.println("创建单个文件" + destFileName + "失败！" + e.getMessage());
            return false;
        }
    }
   
	/**
	 * 创建目录
	 * @param destDirName
	 * @return
	 */
    public static boolean createDir(String destDirName) {
        File dir = new File(destDirName);
        if (dir.exists()) {
            System.out.println("创建目录" + destDirName + "失败，目标目录已经存在");
            return false;
        }
        if (!destDirName.endsWith(File.separator)) {
            destDirName = destDirName + File.separator;
        }
        //创建目录
        if (dir.mkdirs()) {
            System.out.println("创建目录" + destDirName + "成功！");
            return true;
        } else {
            System.out.println("创建目录" + destDirName + "失败！");
            return false;
        }
    }
   
    /**
     * 创建临时文件
     * @param prefix 文件名
     * @param suffix	文件类型
     * @param dirName	路径
     * @return
     */
    public static String createTempFile(String prefix, String suffix, String dirName) {
        File tempFile = null;
        if (dirName == null) {
            try{
                //在默认文件夹下创建临时文件
                tempFile = File.createTempFile(prefix, suffix);
                //返回临时文件的路径
                return tempFile.getCanonicalPath();
            } catch (IOException e) {
                e.printStackTrace();
                System.out.println("创建临时文件失败！" + e.getMessage());
                return null;
            }
        } else {
            File dir = new File(dirName);
            //如果临时文件所在目录不存在，首先创建
            if (!dir.exists()) {
                if (!FileKit.createDir(dirName)) {
                    System.out.println("创建临时文件失败，不能创建临时文件所在的目录！");
                    return null;
                }
            }
            try {
                //在指定目录下创建临时文件
                tempFile = File.createTempFile(prefix, suffix, dir);
                return tempFile.getCanonicalPath();
            } catch (IOException e) {
                e.printStackTrace();
                System.out.println("创建临时文件失败！" + e.getMessage());
                return null;
            }
        }
    }
    
    /** 
     * 获取扩展名 
     * @param fileName 
     * @return 
     */  
    public static String getExtension(String fileName) {  
        if (StringKit.INDEX_NOT_FOUND == StringKit.indexOf(fileName, "."))  
            return StringKit.EMPTY;  
        String ext = StringKit.substring(fileName,  
                StringKit.lastIndexOf(fileName, "."));  
        return StringKit.trim(ext);  
    }
    
    /**
     * 获取文件类型
     * @param fileUrl
     * @return
     */
    public static String getMimeType(String fileUrl) {  
    	    String type = "未知";  
    	    URL u;
			try {
				u = new URL(fileUrl);
				URLConnection uc = null;  
	    	    uc = u.openConnection();  
	    	    type = uc.getContentType();  
			} catch (MalformedURLException e) {
				e.printStackTrace();
			} catch (IOException e) {
				e.printStackTrace();
			}  
    	    return type;  
    }  
    
    
}
